# -*- coding: utf-8 -*-
# @Time    : 2021/12/16 4:46 下午
# @Author  : zlt
# @FileName: util_log.py
# @Software: PyCharm

"""
文件说明：
封装logging类，实例化为log
其他文件导入后，可直接使用log实例来生成日志
"""

import logging
import datetime
import time
import os

#项目所在的根目录
BASE_PATH = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
#日志文件所在的文件夹目录
LOG_PATH = BASE_PATH + "/logs"

#如果日志文件夹不存在就新建一个日志文件夹
if os.path.exists(LOG_PATH) == False:
    os.makedirs(LOG_PATH)
else:
    pass


class Logger(object):
    def __init__(self,log_path = LOG_PATH):
        """

        :param log_path: 日志文件目录
        """
        # 文件的命名
        self.log_name = os.path.join(log_path, '%s.log' % time.strftime('%Y-%m-%d'))
        self.logger = logging.getLogger()
        self.logger.setLevel(logging.DEBUG)

        # 日志输出格式
        self.formatter = logging.Formatter('[%(asctime)s] - %(filename)s] - %(levelname)s: %(message)s')

    def __console(self, level, message):

        # 创建一个FileHandler，用于写到本地
        fh = logging.FileHandler(self.log_name, 'a', encoding='utf-8')
        fh.setLevel(logging.DEBUG)
        fh.setFormatter(self.formatter)
        self.logger.addHandler(fh)

        ch = logging.StreamHandler()  # 创建一个StreamHandler,用于输出到控制台
        ch.setLevel(logging.DEBUG)
        ch.setFormatter(self.formatter)
        self.logger.addHandler(ch)

        if level == 'info':
            self.logger.info(message)
        elif level == 'debug':
            self.logger.debug(message)
        elif level == 'warning':
            self.logger.warning(message)
        elif level == 'error':
            self.logger.error(message)

        # 这两行代码是为了避免日志输出重复问题
        self.logger.removeHandler(ch)
        self.logger.removeHandler(fh)

        # 关闭文件
        fh.close()

    def debug(self, message):
        self.__console('debug', message)

    def info(self, message):
        self.__console('info', message)

    def warning(self, message):
        self.__console('warning', message)

    def error(self, message):
        self.__console('error', message)






